import erniebot


# 初始化 Erniebot
def initialize_erniebot():
    erniebot.api_type = 'aistudio'
    erniebot.access_token = '721efe09951433260c4ab297c2830cffc82e2d77'


def errsentence(user_query):
    initialize_erniebot()

    question = f"""
    【要求】
    1.请帮我检查以下句子是否存在病句并指出问题所在。
    2.如果有病句，请提供修改建议。
    3.每个点输出的内容不能超过30个字
    4.每条输出建议内容后面都需要加上"<br>"这个字符串
    
    【待修改句子】
    {user_query}

    严格按格式输出不超过4处修改内容:
    1. `原文`优化为`xxx`
    2. `原文`优化为`xxx`
    3. `原文`优化为`xxx`
    4. `原文`优化为`xxx`
    """

    response = erniebot.ChatCompletion.create(
        model='ernie-3.5',
        messages=[
            {'role': 'user', 'content': question},
        ],
        temperature=0.8
    )

    return response.get_result()

# 调用方式示例（在其他文件中，首先要导入这个函数）
# user_question = input()
# result = errsentence(user_question)
# print(result)
